finishing up code