Removed unnecessary files and cleaned up code so it works!