Tweaking project 1 assignment