Add extra UI for picking up items
Refactor and fix overlay ui