Fixed code so it actually compiles now