Added in controller support and made the level a little better.