Add model, context and classificatoin layers