added my solution to problem set 3