Move query annotations logic to `data()`

[sanjaynagi]

• We should allow users to query based on annotations (pfam, go) within the data() function. This will simplify code in other functions too.