Getting close to a rough draft. Now, only if I had working code