lab 5 modified.... maybe
fix