Pull request

Created tests for menu class, but haven't integrated them into the main generator script yet.