Updated script for additional analyses