Slight change to hw5 for use at home