Updated plots and refactored GUI code

Signed-off-by: Maxim Osipov <maxim.osipov@gmail.com>