adding more UI varios bug fixes