Add logic to check dropdown box