Updates to main plugin class