working on assignment 2 and 3