updating fixing slight bug