Add ability to get entities by type.