Updates

I added some stuff that was missing. I hope you don't mind.