homework solution fixes