rough solution for hw