Cleaned up $.ready function