fixed bug in 6th homework