Sub-nav for product types

I am trying to create sub-nav for a shop but having some difficulty in getting it to do what I want.

I have a menu which displays a list of collections. When a collection is clicked it goes to that collection and the nav expands to show a list of product types under that collection.
If I then click on a product type it then shows all products in for that product type, however the sub-nav list of product types disappears from my menu.

Here is what I have so far:

{% for link in linklists.sub.links %}
{% if link.title == collection.title %}

• {{ link.title }}
  {% else %}
  
• {{ link.title }}
  {% endif %}
  {% if link.title == collection.title %}
  {% for product in collection.products %}
  
  
  
  • {{ product.type | link_to_type }}
  
  
  
  {% endfor %}
  {% endif %}
  

{% endfor %}

Edit: sorry the code thing is playing up, I can't put it in properly. See the code at http://forums.shopify.com/categories/...

Anyone help?