updated a little bit