Pull request

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