Fixed errors when compiling :)