Added some annotations to Python