Got homework working but need to repeat