Clean up code snippet