made design a bit better