Complete homework
Fixing issue