Updating solutions homework 2