added stuff, sorry for bad commit msg but not sure what ive done, it works.