Make python indent smarter