Complete Flagging Tests and functionality