Removed unnecessary files and cleaned up the code