Seems to be working now :) Not sure what I was doing wrong.