For readability's sake, can you change this into if/then/else statements?