Use subset() function for readability. Update comments.