Improved code after watching Assignment Helpers Solution