Added the set option
Use case in place of if