fixing small ui issues