[~] Small changes to UI
Completion of task