First trial with while loops and if statements