Little update for homework 5