Added screenshots and removed empty unit test