Add plotting options to function