Finished simple branch distribution generation
Changed to conform to PEP8