That looks good. Do you know where I can find the code for this or maybe something similar?