Fixing the code so now it works