First commit
Assignment 2 codes