Cleans up files and adds better documentation