README and comments
Added menu functionality