fixed some ui issues