Fix up models for titanic project