Minor bug fixes, add menu