Simplify method

I think this form is more logical.
PS. It passes all tests.