Add template + minors changes