Edited with PyCharm