Fixed up the code. Should be good to go.