How to correctly simplify these functions.