Assignment 1, finished I hope