Added an operator overload

just refactored a tiny bit.