Feature/add setup functionality