Few style fix
Home work 2