new solution 2 fun with if/else