edited it so it works now ok