Add some very basic type inference
Type check return statements