Bug Fixes/UI updates