Completed the if/else assignment