fix
and add display