Bug Fixes + UI Fixes