some minor changes
minor fix