Fix Python style, add comments & TODOs