Pep 8 improvements and code refactoring in drawing pakage