Just need to fix function type checking for statements
Square passes