Added homework solution