With set() and find() now bypassing Square and using Entity directly, should we have get() return List<Entity> ?  That way the Square class won't have to be public, it'll be an implementation detail.