Add screenshot + some code changes.