Updated the build.gradle to make the build work