add more features in code