Updated code to work with ss3.0.3

I've added some minor modifications so your code works with SilverStripe 3.0.3
Would you like me to fork and push so others can use it also?