complete assignment, want to fix the bug before emailing