Add test case where selected feature are only available on some plans.
Refactor code for clarity.