Fixed small asthetic changes