added homework, should be working