Moved Code Back

Moved code back to where it was originally.