Adds more type checking