added code and edited design