Clean up code in all files