Format code a bit