Small edit. Found some deprecated code that needed axing.