getting closer.
don't forget to modify template logic