Small changes
Done