Added UI

Added a text based interface to control the generation.