added new feature
fixed something funny!