Fixed messy code.