Bug fixes / display fixes