Fix #9
Update UI