Changed design to look similar to the original one