Added weighted graph for better AI