Small fixes i think