Fix positioning + legend dimensions