Added function for simple plotting