Shorten Problem 1 Solution

After some awesome feedback from my buddy Steve