Update class-functionality-plugin.php