Add descriptions, make python code much simpler