Bug fixed - Class not compiling