added some cleanup that i forgot