Assignment 1 for comp308