fixed a couple functions that they gave us