Working on adding the flagging feature.