Improved plotting scripts - for all studies