Updated build.gradle for version 0.5.4