fix bugs
ADD TEST DATA
fix bugs