added logic. Todo: add people input