Python Function Specify Input Type