Clean up code by replacing imports and references of `_ast` to `ast`.