fixes after watching solutions