changed UI to look a bit better