Fixing some issues
Adding a new feature