Add psets 1-3

Code for problem sets 1, 2 and 3