Debug/small fixes