small fixes for the source code