Changed it to suit my needs