Fixed small little problem on line 64 of code