Updated the repository to match my working version