Updates!

Cleaned up your initial code, great job!!! 
Added a bit of functionality!