Some little changes

Fix #4