Make code PEP8 compliant
Add simple test for an assignment