Update main.h

add template <class TypeArray>