Add code back and cleaned up files