Worked on exercise 2
New problem solved