Add code to prepare preliminary plots