Cleaned up code and added more comments. Also renamed Piece folder to piece_images.