Final tweaks
Moar fixes