some refactoring
add screenshot to readme