Added some awesome logic.