Refactored my if/else statement to use my new friend, the ternary operator