Fixes #17
Type hint container

shouldn't be using the container like this, and will probably change to setContainer() method