Minor bugs fixed

Minor bugs with display and directories fixed