No need the alpha toggle in make_png
- Dimension can be obtained by data.shape[2]
- Check for dimension and data type
- Add alpha toggle to _screenshot