minor changes. Didn't have time to do much