Made changes to the UI, minor fixes.