edit to include class instead of select