Fixed bug so graph now works with all valid sizes