finish python rough draft