Almost done with Assignment 3