Clean up code before publishing