Some great additions to your code. Tell me if you like it :D