Part a complete, working on part b