[TASK] Add unit dropdown - Closes #9