Fixed code, compiles now