Refined it a little bit