Fixing silly mistakes made by me :)