clean up some of the code
add screenshot