Updated class refference for add_action