Improved code

![beaslutdowhateveryouwant](https://f.cloud.github.com/assets/3579182/2272426/d9262822-9f04-11e3-8e1c-11642607583e.png)