Add missing include for math.h