Cleaned up some comments and added screenshot