Made code more better. Don't you think?

Just a demo.