Shorter code

This is a shorter code solution.