Is your code for the second part working?
I'm not quite sure what you are asking.