edited a little bit