Edited Simon code. Should be working now?