Add if-statement checking
Type Inference progress..