Add code for titanic competition