Even though I don't quite understand your issue, I have done something. It is extremely simple and I haven't largely changed the code.
