Add simple type checker