Use HOME instead of hardcoding the directory