Made some minor ammedments to the UI views