Question 2 Revision 3

Using subset function instead