Fix a ton of bugs