minor refactor. screenshot update