Add the support for screenshot

Signed-off-by: Connor Yan <126ium@gmail.com>