Fixed a bug cause Nick said to
added some filthy stuff