Use python2
Improve formatting