Minor changes

- close connection after getting all data