Add mock data feature and fix ui