Moved some functionality to setUp method.