Cleaned up and added a note