updated with future direction in the first model. Don't have time to fix it now.