small fixes for code