Simplified the build.gradle file a bit