changes made
Fix bug