Updated Code

I updated some of the code. Let me know what you think. -Parag