Simple Python code/AST formater