making assignment 1 prettier