More exercise, loops and functions, if else statements