Improve work with python