Add small fixes