better code - Level 4 Works - After Level 4 WIP