Finished up the assignment!