Bug fixes and improved UI functionality.