Code Edit

Refactored some code. That is all.