Went through the code with PEP 8 formatting rules (placement of columns). Just for the sake of beauty:-)