some changes for better visualisation