add proper type hint for container

Allows PHPStorm to lookup the available methods.