added small fixes