Best way to go about setting up the logic?