Moved add_action to bottom of source code