Small changes to make code completely work