Tidy + fixes