Fix bugs and update from Joukahainen