Using a directive worked out really nice