Adding more tests and fixing issues with the compiler