Enhance gradle build script