problem 4
problem 5
changed naming convention