small refactoring
add screenshot