fixed a bug
further development