Use logic from model