add display method and make some other change