Moving plugin class to build.gradle