changed input location
added solution to problem 5