Forgot to add some stuff. Fixed

Hope it's all now.