Cleaned up code directory