Refactored.
Added mock data.