Added task wrapper class in build.gradle