Updated Code

Might meet extra credit standards....