Fixed some stuff

Added some stuff that I think you forgot.