Add in flag functionality.