almost there... got a bug in display