add screenshots; code cleanup :P