Cleaned up the code folder