implemented github issue #6
fixed minor ui bug
fixed minor ui bug