Updated Titanic prediction code