Added some code. Almost got something working.