correct python format and additional comments