Added a video and refactored the code