Fix #4 - add basic menu