improved the look a bit more