Fixed a few bugs, it works great now!