Add function to show a check mark or an X