Forgot to add 'dispose' method