Clean this code up a wee bit