Fix comparison for unit literals
Add checking for main :: (Unit -> Unit) function in intepreter mode