got if/else working