added solution to problem2