Changed the code a bit