Fixed some stuff

Got an error, not sure what changes were made, but it's working.