Adding ability to use custom grid template