Add argument type check

More checks will be added soon.