Fixed missing file
Added small changes