Yaser, my spidey-sense indicates you have been bamboozled. You went above and beyond the call. A demo task should provide the employer a basic sense of your abilities. The word demo implies a demonstration, not a full working implementation. You should not need to "improve your implementation" because they "believe in people". Take it as a lesson learned.
We're a place where coders share, stay up-to-date and grow their careers.
We strive for transparency and don't collect excess data.