Small fixes

Will do. I didn't realize that you could do that in python.