Changed the code (Slightly)

-