Modify edit to include screenshot