Fixed Tom's code. Duh