HomeMade Improvement from CodeWithChris assignment