Updated assignement 2