Add include for function, simplify function calls