Update gradle build. Fixes #1