added plotting features