Assignment 3 almost done!