fixed bug, minor ui update