Updated screenshots and removed unneccessary imports.