Add proper template