First answer to problem set 1