Assignment 2 by me