The code flow has been neatened up. It has also been pylinted. Purpose of code review is for it to be checked.