Add main element template + fix tree regression