- Made corrections to the ui