Patch 2

fixing ugs