Small modification to map drawing