Added missing include statement to partition.h