new code

Elana Shi helped with some code problems