Code beautification

Thank you, pyCharm :-)