Started working on problem set 1