Assignment on reinforcement learning from machine learning course at KTH