Made code more better. 

Just a demo.