minor changes to UI, added GUI diagram