Format code properly