Added assumptions and TODO in comment for `atan2`