Made some changes, thanks Mineothor on minecraftforge.net